import tensorflow as tf

class BPMEstimator(tf.keras.Model):
    def __init__(self):
        super().__init__()
        self.lstm = tf.keras.layers.LSTM(128, return_sequences=True)
        self.regressor = tf.keras.layers.Dense(1)

    def call(self, inputs):
        return self.regressor(self.lstm(inputs))